package com.xinqi.common.mybatis.data.repo;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Constants;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.xinqi.common.base.response.PageInfo;
import com.xinqi.common.mybatis.utils.Pages;
import org.apache.ibatis.annotations.Param;

import java.io.Serializable;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * @author linkai
 */
public interface SuperMapper<T> extends BaseMapper<T> {

    /**
     * 通过id全字段更新
     */
    int updateAllColumnById(@Param(Constants.ENTITY) T entity);

    /**
     * 根据 ID 查询数据, 无视逻辑删除字段
     *
     * @param id 主键ID
     */
    T selectByIdWithDeleted(Serializable id);


    /**
     * 根据 ID 查询
     */
    default <R> R selectDtoById(Serializable id, Function<T, R> convert) {
        T obj = this.selectById(id);
        if (ObjectUtil.isNull(obj)) {
            return null;
        }
        return convert.apply(obj);
    }

    /**
     * 查询（根据ID 批量查询）
     */
    default <R> List<R> selectDtoBatchIds(Collection<? extends Serializable> idList, Function<T, R> convert) {
        List<T> list = this.selectBatchIds(idList);
        if (CollUtil.isEmpty(list)) {
            return CollUtil.newArrayList();
        }
        return list.stream().map(convert).collect(Collectors.toList());
    }

    /**
     * 查询（根据 columnMap 条件）
     */
    default <R> List<R> selectDtoByMap(Map<String, Object> map, Function<T, R> convert) {
        List<T> list = this.selectByMap(map);
        if (CollUtil.isEmpty(list)) {
            return CollUtil.newArrayList();
        }
        return list.stream().map(convert).collect(Collectors.toList());
    }

    /**
     * 根据 entity 条件，查询一条记录
     */
    default <R> R selectDtoOne(Wrapper<T> wrapper, Function<T, R> convert) {
        T obj = this.selectOne(wrapper);
        if (ObjectUtil.isNull(obj)) {
            return null;
        }
        return convert.apply(obj);
    }

    /**
     * 根据 entity 条件，查询全部记录
     */
    default <R> List<R> selectDtoList(Wrapper<T> wrapper, Function<T, R> convert) {
        List<T> list = this.selectList(wrapper);
        if (CollUtil.isEmpty(list)) {
            return CollUtil.newArrayList();
        }
        return list.stream().map(convert).collect(Collectors.toList());
    }

    /**
     * 分页查询VO
     */
    default <R, P extends IPage<R>> P selectDtoPage(IPage<T> page, Wrapper<T> wrapper, Function<T, R> convert) {
        IPage<T> pageData = this.selectPage(page, wrapper);
        IPage<R> voPage = new Page<>(pageData.getCurrent(), pageData.getSize(), pageData.getTotal());
        if (CollUtil.isEmpty(pageData.getRecords())) {
            return (P) voPage;
        }
        voPage.setRecords(pageData.getRecords().stream().map(convert).collect(Collectors.toList()));
        return (P) voPage;
    }

    /**
     * 分页查询VO
     */
    default <R> PageInfo<R> selectDtoPageInfo(IPage<T> page, Wrapper<T> wrapper, Function<T, R> convert) {
        IPage<T> pageData = this.selectPage(page, wrapper);
        return Pages.convert(pageData, convert);
    }

}
